# 这是一个示例 Python 脚本。

# 按 Shift+F10 执行或将其替换为您的代码。
# 按 双击 Shift 在所有地方搜索类、文件、工具窗口、操作和设置。


import os

from settings import WcSettings
from filens import get_workbook, log_file

etf_list = list(set(WcSettings.ETF_LIST))
url = WcSettings.GET_URL

filen_list = []
data = []


def processing_data(url):
    filename_list = os.listdir(url)

    for item in filename_list:
        if len(item.split('.')) > 1 and item.split('.')[-1] in ['xlsx', 'xls']:
            data.append(os.path.join(url, item))

        elif len(item.split('.')) == 1:
            if os.path.join(url, item) not in filen_list:
                filen_list.append(os.path.join(url, item))
        else:
            continue


def main():
    processing_data(url)

    for item in filen_list:
        processing_data(item)

    for item in data:
        get_workbook(item)

    log_file(WcSettings.ETF_LIST)


if __name__ == '__main__':
    main()
